How can I correctly convert a list to a numpy array in my TensorFlow model